"""
利用装饰器强制函数上的类型检查

问题：
    作为某种编程规约，你想在对函数参数进行强制类型检查。
解决方案：
    使用装饰器技术来实现 @typeassert。
"""

from inspect import signature
from functools import wraps


def typeassert(*ty_args, **ty_kwargs):
    """
    类型检查断言装饰器
    """

    def decorate(func):
        if not __debug__:
            return func

        sig = signature(func)
        bound_types = sig.bind_partial(*ty_args, **ty_kwargs).arguments

        @wraps(func)
        def wrapper(*args, **kwargs):
            bound_values = sig.bind(*args, **kwargs)
            for name, value in bound_values.arguments.items():
                if name in bound_types:
                    if not isinstance(value, bound_types[name]):
                        raise TypeError(
                            "Argument {} must be {}".format(name, bound_types[name])
                        )
            return func(*args, **kwargs)

        return wrapper

    return decorate


@typeassert(int, z=int)
def say(x, y, z):
    print(x, y, z)


def use_type_assert():
    say(1, 2, 3)
    # 输出结果：
    """
    1 2 3
    """

    say(1, "Kevin", 3)
    # 输出结果：
    """
    1 Kevin 3
    """

    say(1, "Kevin", "KK")
    # 输出结果：
    """
    TypeError: Argument z must be <class 'int'>
    """


if __name__ == '__main__':
    use_type_assert()
